package com.vca.common.interceptor;

import org.springframework.beans.factory.annotation.Value;
import org.springframework.core.io.Resource;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.core.io.support.ResourcePatternResolver;
import org.springframework.util.AntPathMatcher;
import org.springframework.util.FileCopyUtils;
import org.springframework.web.servlet.handler.HandlerInterceptorAdapter;
import sun.misc.BASE64Decoder;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.PrintWriter;

/**
 * Swagger 文档
 */
public class SwaggerInterceptor extends HandlerInterceptorAdapter {
    private String username;
    private String password;
    private Boolean check;

    private Boolean enable;

//    String ededn;

    public SwaggerInterceptor(String username, String password, Boolean check, Boolean enable) {
        this.username = username;
        this.password = password;
        this.check = check;
        this.enable = enable;
    }

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        if (!enable) {
            String requestURI = request.getRequestURI();
            if (requestURI.contains("/swagger") || requestURI.contains("/v2/api-docs") || requestURI.contains("actuator") || requestURI.contains("webjars")) {
                response.setStatus(HttpServletResponse.SC_NOT_FOUND);
                return false;
            }
            return true;
        } else {
            String authorization = request.getHeader("Authorization");
            boolean isAuthSuccess = httpBasicAuth(authorization);
            if (!isAuthSuccess) {
                response.setCharacterEncoding("utf-8");
                response.setStatus(401);
                response.setHeader("WWW-authenticate", "Basic realm=\"Realm\"");
                try (PrintWriter writer = response.getWriter()) {
                    writer.print("Forbidden, unauthorized user");
                }
            }
            return isAuthSuccess;
        }
    }

    public boolean httpBasicAuth(String authorization) throws IOException {
        if (check) {
            if (authorization != null && authorization.split(" ").length == 2) {
                String userAndPass = new String(new BASE64Decoder().decodeBuffer(authorization.split(" ")[1]));
                String username = userAndPass.split(":").length == 2 ? userAndPass.split(":")[0] : null;
                String password = userAndPass.split(":").length == 2 ? userAndPass.split(":")[1] : null;
                return this.username.equals(username) && this.password.equals(password);
            }
            return false;
        }
        return true;
    }

    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {
//        String uri = request.getRequestURI();
//        AntPathMatcher pathMatcher = new AntPathMatcher();
//        if (!pathMatcher.match("/swagger-ui.html", uri) && !pathMatcher.match("/webjars/**", uri)) {
//            response.setStatus(404);
//            return;
//        }
//        ResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
//        Resource[] resources = resolver.getResources("classpath:/META-INF/resources" + uri);
//        if (resources.length > 0) {
//            FileCopyUtils.copy(resources[0].getInputStream(), response.getOutputStream());
//        } else {
//            response.setStatus(404);
//        }
    }
}
